from typing import List
from pydantic import BaseModel


class BBox(BaseModel):
    xmin: int
    ymin: int
    xmax: int
    ymax: int
    label: str


class ImageAnnotation(BaseModel):
    filename: str
    width: int
    height: int
    bboxes: List[BBox]
